'''
draw a graph of agent using the porter stemming algorithm and eliminating stop 
words.
of accuracy opposed to number of features.
'''

from agent1 import Agent1
from one_agent_accurcy import AgentTester
from text_example import createRealDatasets
from numpy import arange

datasets = createRealDatasets()
results = {}
sizes = arange(start=20,stop = 60, step=20)
    
for name, data in datasets.items():
    
    results[name] = {}
            
    for number_of_features in sizes:
        
        class a(Agent1):
            pass
        a.set(number_of_features)
        tester = AgentTester()
        r = tester.run(data, a, 1, 30, 10)
        results[name][number_of_features] = r
        
        
print results